# JellyBeanColourCounter.py
#
# Description: This program takes an image as input and outputs
#              the percentage of each jellybean colour in the image.
#
# Author: AL and AL
# Date: March 2024

# Import image processing libraries
from PIL import Image

# Import myColourModule
import myColourModule


# ***Main part of the program

# Open an image of jellybeans and load its bitmap
jellyBeansOpen = Image.open("jelly_beans.jpg")
jellyBeansLoad = jellyBeansOpen.load()

# Create a list to store the pixels that are yellow
yellowPixelList = []

# Get dimensions of bitmap
width = jellyBeansOpen.width
height = jellyBeansOpen.height

# Go through all the pixels in the image
for j in range(height):
  for i in range(width):
    r = jellyBeansLoad[i,j][0]
    g = jellyBeansLoad[i,j][1]
    b = jellyBeansLoad[i,j][2]

    # If it's yellow, then add that pixel to the yellow list
    # if myColourModule.isPixelYellow(r, g, b):
    if myColourModule.isPixelYellow((r, g, b)):
      #yellowPixelList.append(jellyBeansLoad[0,0]) <- typo!
      yellowPixelList.append(jellyBeansLoad[i,j])

# Get the length of the yellow pixel list 
numberOfYellowPixels = len(yellowPixelList)

# Compute the total number of pixels in the image
numberOfTotalPixels = height * width

print(f'numberOfYellowPixels: {numberOfYellowPixels}')
print(f'numberOfTotalPixels: {numberOfTotalPixels}')

# Close the file
jellyBeansOpen.close()